#include"tools.h"
#include<assert.h>
#include<getch.h>
#include<string.h>
#include<stdbool.h>
char get_cmd(char start,char end)
{
	assert(start <= end);

	puts("-------------------");
	printf("请输入指令：");
	for(;;)
	{
		char cmd = getch();
		if(start <= cmd && cmd <= end)
		{
			printf("%c\n",cmd);
			return cmd;
		}
	}
}

void print_sec(const char* msg,float sec)
{
	printf("\33[01;32m %s \33[00m\n",msg);
	usleep(1000000*sec);
}

void anykey_continue(void)
{
	puts("按任意键继续...");
	getch();
}

char* get_str(char* str,size_t size)
{
	assert(NULL!=str && size>1);

	// 计算fgets读取了多少个字符
	size_t len = strlen(fgets(str,size,stdin));

	// 如果最后一个字符是'\n'，则把它改为'\0'
	if('\n' == str[len-1])
		str[len-1] = '\0';
	else
		// 如果最后一个字符不是'\n'，则说明缓冲区中有垃圾数据，则需要清理输入缓冲区
		while('\n' != getchar());
	return str;
}

char* get_passwd(char* passwd,size_t size)
{
	int i = 0;
	while(i < size-1)
	{
		passwd[i] = getch();

		// 读取到退格键
		if(127 == passwd[i])
		{
			// 数组中有已输入密码
			if(i > 0)
			{
				// 删除一位密码
				i--;
				printf("\b \b");
			}
			continue;
		}
		
		i++;
		printf("*");
	}
	passwd[size-1] = '\0';
	printf("\n");
	return passwd;
}

bool yes_or_no(void)
{
	printf("是否确认此操作(y/n)?");
	for(;;)
	{
		char cmd = getch();
		if('y' == cmd || 'Y' == cmd || 'n' == cmd || 'N' == cmd)
		{
			printf("%c\n",cmd);
			return 'y'==cmd || 'Y'==cmd;
		}
	}
}
